package com.tom.demo.component;

import cn.hutool.core.util.IdUtil;
import com.tom.demo.common.config.MinioConfig;
import io.minio.*;
import io.minio.errors.*;
import io.minio.http.Method;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.io.IOException;
import java.io.InputStream;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.util.Objects;
import java.util.concurrent.TimeUnit;

/**
 * minio上传下载的服务组件
 */
@Slf4j
@Component
public class MinioComponent {
    @Resource
    private MinioClient minioClient;
    @Resource
    private MinioConfig minioConfig;

    /**
     * 创建桶(如果),传入要创建的桶的名字
     * @param bucketName
     * @throws Exception
     */
    public void createBucket(String bucketName) throws Exception{
        if(!bucketName.matches("^[a-z0-9][a-z0-9\\\\-]+[a-z0-9]$")){
            //桶的名称不符合规则
            log.error("<=============桶的名称不符合规则============>");
            return;
        }
        //判断这个桶是否已经存在，如果不存在会自动创建出来
        if (!minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucketName).build())) {
            //1:先创建桶
            minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucketName).build());
        }
    }

    /**
     * 生成一个7天过期的临时访问地址 (IO 的签名 URL 有一个严格的过期时间限制：最短 1 秒，最长 7 天（即 604 800 秒）)
     * 返回的是一个可以用来直接访问文件的预签名URL
     * 注意:该路劲是可以访问的，但是不能直接下载，同时桶的权限可以是私有的private
     * @param bucketName
     * @param file
     * @return
     */
    public String uploadAndGetResignedUrl7Days(
            String bucketName,
            MultipartFile file
    ) throws Exception {
        // 1. 生成文件名
        String suffix = Objects.requireNonNull(file.getOriginalFilename())
                .substring(file.getOriginalFilename().lastIndexOf("."));
        String objectName = IdUtil.simpleUUID() + suffix;

        // 2. 上传到私有桶
        minioClient.putObject(
                PutObjectArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .stream(file.getInputStream(), file.getSize(), -1)
                        .contentType(file.getContentType())
                        .build()
        );

        // 3. 生成一个 GET 的 presigned URL（有效期 7 天）
        return minioClient.getPresignedObjectUrl(
                GetPresignedObjectUrlArgs.builder()
                        .bucket(bucketName)
                        .object(objectName)
                        .method(Method.GET)
                        .expiry(7, TimeUnit.DAYS)
                        .build()
        );
    }

    /**
     * 传入要上传的桶的名字和文件类容
     * 返回一个可以直接使用的地址链接
     * @param bucketName
     * @param file
     * @return
     * @throws Exception
     */
    public String uploadFile(String bucketName, MultipartFile file) throws Exception {
        //生成一个新的文件名字
        String fileName = IdUtil.simpleUUID()+ Objects.requireNonNull(file.getOriginalFilename()).substring(file.getOriginalFilename().lastIndexOf("."));
        minioClient.putObject(PutObjectArgs.builder()
                .bucket(bucketName)
                .object(fileName)
                .stream(file.getInputStream(), file.getInputStream().available(), -1)
                .contentType(file.getContentType())
                .build());
        return minioConfig.getEndpoint() + "/" + bucketName + "/" + fileName;
    }

    /**
     * 下载文件(返回的是一个二进制流)
     * @param bucketName
     * @param filename
     * @return
     * @throws Exception
     */
    public InputStream downloadFile(String bucketName, String filename) throws Exception {
        return minioClient.getObject(GetObjectArgs.builder()
                .bucket(bucketName)
                .object(filename)
                .build());
    }



}